package com.kk.common;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.Data;
import java.io.Serializable;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 分页响应对象
 *
 * @author kk
 */
@Data
public class PageResult<T> implements Serializable {

    private static final long serialVersionUID = 1L;

    /**
     * 数据列表
     */
    private List<T> records;

    /**
     * 总记录数
     */
    private Long total;

    /**
     * 当前页码
     */
    private Long current;

    /**
     * 每页显示条数
     */
    private Long size;

    /**
     * 总页数
     */
    private Long pages;

    public PageResult() {
    }

    public PageResult(List<T> records, Long total, Long current, Long size) {
        this.records = records;
        this.total = total;
        this.current = current;
        this.size = size;
        this.pages = (total + size - 1) / size;
    }

    /**
     * 从MyBatis-Plus的Page对象转换（类型相同）
     *
     * @param page MyBatis-Plus分页对象
     * @param <T>  数据类型
     * @return PageResult对象
     */
    public static <T> PageResult<T> of(Page<T> page) {
        return new PageResult<>(
                page.getRecords(),
                page.getTotal(),
                page.getCurrent(),
                page.getSize()
        );
    }

    /**
     * 从MyBatis-Plus的Page对象转换（类型转换）
     *
     * @param page      MyBatis-Plus分页对象
     * @param converter 类型转换函数
     * @param <S>       源类型（如 SysUser）
     * @param <T>       目标类型（如 UserListVO）
     * @return PageResult对象
     */
    public static <S, T> PageResult<T> of(Page<S> page, Function<S, T> converter) {
        List<T> records = page.getRecords().stream()
                .map(converter)
                .collect(Collectors.toList());

        return new PageResult<>(
                records,
                page.getTotal(),
                page.getCurrent(),
                page.getSize()
        );
    }

}